package com.ruoyi.web.core.config;

import com.ruoyi.common.config.RuoYiConfig;
import io.swagger.annotations.ApiOperation;
import io.swagger.models.auth.In;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Profile;
import springfox.documentation.builders.ApiInfoBuilder;
import springfox.documentation.builders.PathSelectors;
import springfox.documentation.service.*;
import springfox.documentation.spi.DocumentationType;
import springfox.documentation.spi.service.contexts.SecurityContext;
import springfox.documentation.spring.web.plugins.Docket;

import java.util.Collections;
import java.util.List;

/**
 * Swagger2的接口配置
 * 
 * @author ruoyi
 */
//@Configuration
//public class SwaggerConfig
//{
//    /** 系统基础配置 */
//    @Autowired
//    private RuoYiConfig ruoyiConfig;
//
//    /** 是否开启swagger */
//    @Value("${swagger.enabled}")
//    private boolean enabled;
//
//    /** 设置请求的统一前缀 */
//    @Value("${swagger.pathMapping}")
//    private String pathMapping;
//
//    /**
//     * 创建API
//     */
//    @Bean
//    public Docket createRestApi()
//    {
//        return new Docket(DocumentationType.OAS_30)
//                // 是否启用Swagger
//                .enable(enabled)
//                // 用来创建该API的基本信息，展示在文档的页面中（自定义展示的信息）
//                .apiInfo(apiInfo())
//                // 设置哪些接口暴露给Swagger展示
//                .select()
//                // 扫描所有有注解的api，用这种方式更灵活
//                .apis(RequestHandlerSelectors.withMethodAnnotation(ApiOperation.class))
//                // 扫描指定包中的swagger注解
//                .apis(RequestHandlerSelectors.basePackage("com.ruoyi.web.controller.storeSettings"))
//                // 扫描所有 .apis(RequestHandlerSelectors.any())
//                .paths(PathSelectors.any())
//                .build()
//                /* 设置安全模式，swagger可以设置访问token */
//                .securitySchemes(securitySchemes())
//                .securityContexts(securityContexts())
//                .pathMapping(pathMapping);
//    }
//
//    /**
//     * 安全模式，这里指定token通过Authorization头请求头传递
//     */
//    private List<SecurityScheme> securitySchemes()
//    {
//        List<SecurityScheme> apiKeyList = new ArrayList<SecurityScheme>();
//        apiKeyList.add(new ApiKey("Authorization", "Authorization", In.HEADER.toValue()));
//        return apiKeyList;
//    }
//
//    /**
//     * 安全上下文
//     */
//    private List<SecurityContext> securityContexts()
//    {
//        List<SecurityContext> securityContexts = new ArrayList<>();
//        securityContexts.add(
//                SecurityContext.builder()
//                        .securityReferences(defaultAuth())
//                        .operationSelector(o -> o.requestMappingPattern().matches("/.*"))
//                        .build());
//        return securityContexts;
//    }
//
//    /**
//     * 默认的安全上引用
//     */
//    private List<SecurityReference> defaultAuth()
//    {
//        AuthorizationScope authorizationScope = new AuthorizationScope("global", "accessEverything");
//        AuthorizationScope[] authorizationScopes = new AuthorizationScope[1];
//        authorizationScopes[0] = authorizationScope;
//        List<SecurityReference> securityReferences = new ArrayList<>();
//        securityReferences.add(new SecurityReference("Authorization", authorizationScopes));
//        return securityReferences;
//    }
//
//    /**
//     * 添加摘要信息
//     */
//    private ApiInfo apiInfo()
//    {
//        // 用ApiInfoBuilder进行定制
//        return new ApiInfoBuilder()
//                // 设置标题
//                .title("标题：若依管理系统_接口文档")
//                // 描述
//                .description("描述：用于管理集团旗下公司的人员信息,具体包括XXX,XXX模块...")
//                // 作者信息
//                .contact(new Contact(ruoyiConfig.getName(), null, null))
//                // 版本
//                .version("版本号:" + ruoyiConfig.getVersion())
//                .build();
//    }

/**
 * Swagger2的接口配置（优化版）
 */
@Configuration
@Profile({"dev", "test"}) // 只在开发和测试环境启用
public class SwaggerConfig
{
    @Autowired
    private RuoYiConfig ruoyiConfig;

    @Value("${swagger.enabled}")
    private boolean enabled;

    @Value("${swagger.pathMapping}")
    private String pathMapping;

    @Bean
    public Docket createRestApi() {
        return new Docket(DocumentationType.OAS_30)
                .enable(enabled)
                .apiInfo(apiInfo())
                .select()
                // 优化扫描策略，先扫描方法注解再扫描包
                .apis(input -> {
                    // 优先检查方法是否有ApiOperation注解
                    if (input.isAnnotatedWith(ApiOperation.class)) {
                        return true;
                    }
                    // 其次检查是否在指定包内
//                    return input.getHandlerMethod().getBeanType().getName()
//                            .startsWith("com.ruoyi.web.controller.storeSettings");
                    String beanPackage = input.getHandlerMethod().getBeanType().getName();
                    return beanPackage.startsWith("com.ruoyi.web.controller") ||
                            beanPackage.startsWith("com.ruoyi.system");
                })
                .paths(PathSelectors.any())
                .build()
                .securitySchemes(securitySchemes())
                .securityContexts(securityContexts())
                .pathMapping(pathMapping);
    }

    private List<SecurityScheme> securitySchemes() {
        return Collections.singletonList(
                new ApiKey("Authorization", "Authorization", In.HEADER.toValue()));
    }

    private List<SecurityContext> securityContexts() {
        return Collections.singletonList(
                SecurityContext.builder()
                        .securityReferences(defaultAuth())
                        // 优化路径匹配，排除基础路径
                        .operationSelector(o -> !o.requestMappingPattern().startsWith("/error") &&
                                !o.requestMappingPattern().startsWith("/actuator"))
                        .build());
    }

    private List<SecurityReference> defaultAuth() {
        AuthorizationScope authorizationScope = new AuthorizationScope("global", "accessEverything");
        return Collections.singletonList(
                new SecurityReference("Authorization", new AuthorizationScope[]{authorizationScope}));
    }

    private ApiInfo apiInfo() {
        return new ApiInfoBuilder()
                .title(ruoyiConfig.getName() + "_接口文档")
                .description("门店设置模块接口文档")
                .contact(new Contact(ruoyiConfig.getName(), null, null))
                .version("版本号: " + ruoyiConfig.getVersion())
                .build();
    }
}


